# -*- coding: utf-8 -*-
"""
@Time ： 2021/10/24 22:34
@Auth ： Mr.Kd 1548826452
@Company ：特斯汀学院 @testingedu.com.cn
@Function ：请输入模块功能描述
"""
import os
import xlrd

# from common.Logger import logger


class Reader:
    def __init__(self):
        self.workbook=None
        self.sheet=None
        self.rows=0
        self.r=0
    def openexcel(self, srcfile=''):
        xlrd.Book.encoding = "utf8"
        # 打开excel文件
        self.workbook = xlrd.open_workbook(filename=srcfile)
        # print(self.workbook)
        return self.workbook
    def get_sheets(self):
        # 获取所有sheet的名字，并返回为一个列表
        self.sheets = self.workbook.sheet_names()

        return self.sheets
    def set_sheets(self,name):
        #读取sheet名为name的sheet表
        self.sheet=self.workbook.sheet_by_name(name)
        #读取这个表的行数
        self.rows=self.sheet.nrows
        self.r=0
        return
    def readline(self):
        lines = []
        #如果当前还没到最后一行，则往下读取一行
        lines=[]
        for i in range(self.rows):
            line=[]
            row=self.sheet.row_values(i)
            k=0
            for cell in row:
                row[k]=str(cell)
                k=k+1
                if cell is None:
                    line.append('')
                else:
                    line.append(str(cell))
            # print(line)
            lines.append(line)
        return lines

if __name__ == '__main__':
    reader = Reader()
    # 打开一个excel

    # reader.set_sheets('test1')
    # reader.readline()







